{% extends 'base.html' %}
{% load static %}

{% block title %}Model Comparison - NEPSE Stock Forecasting{% endblock %}

{% block content %}
<div class="px-4 py-6 sm:px-0">
    <!-- Header -->
    <div class="mb-8">
        <h1 class="text-3xl font-bold text-gray-900">Model Performance Comparison</h1>
        <p class="mt-2 text-gray-600">Compare LSTM and SVR model performance across different stocks</p>
    </div>

    <!-- Stock Selection -->
    <div class="bg-white shadow rounded-lg p-6 mb-8">
        <h3 class="text-lg font-medium text-gray-900 mb-4">Select Stock for Comparison</h3>
        
        <div class="flex space-x-4">
            <div class="flex-1">
                <label for="comparisonStock" class="block text-sm font-medium text-gray-700">Stock Symbol</label>
                <select id="comparisonStock" class="mt-1 block w-full border-gray-300 rounded-md shadow-sm focus:ring-blue-500 focus:border-blue-500">
                    <option value="">Select a stock...</option>
                    {% for stock in stocks %}
                        <option value="{{ stock.symbol }}">{{ stock.symbol }} - {{ stock.company_name }}</option>
                    {% endfor %}
                </select>
            </div>
            
            <div class="flex items-end">
                <button id="loadComparison" class="bg-blue-600 text-white py-2 px-6 rounded-md hover:bg-blue-700 focus:outline-none focus:ring-2 focus:ring-blue-500">
                    <i class="fas fa-chart-bar mr-2"></i>
                    Load Comparison
                </button>
            </div>
        </div>
    </div>

    <!-- Model Metrics Comparison -->
    <div id="comparisonResults" class="hidden">
        <!-- Performance Metrics Cards -->
        <div class="grid grid-cols-1 md:grid-cols-2 gap-6 mb-8">
            <!-- LSTM Metrics -->
            <div class="bg-white shadow rounded-lg p-6">
                <div class="flex items-center mb-4">
                    <div class="flex-shrink-0">
                        <i class="fas fa-brain text-blue-600 text-2xl"></i>
                    </div>
                    <div class="ml-3">
                        <h3 class="text-lg font-medium text-gray-900">LSTM Neural Network</h3>
                        <p class="text-sm text-gray-500">Deep Learning Model</p>
                    </div>
                </div>
                
                <div class="grid grid-cols-2 gap-4">
                    <div class="bg-blue-50 p-3 rounded-lg">
                        <div class="text-xs font-medium text-blue-600 uppercase tracking-wider">RMSE</div>
                        <div class="text-lg font-semibold text-blue-900" id="lstmRmse">-</div>
                    </div>
                    <div class="bg-blue-50 p-3 rounded-lg">
                        <div class="text-xs font-medium text-blue-600 uppercase tracking-wider">R² Score</div>
                        <div class="text-lg font-semibold text-blue-900" id="lstmR2">-</div>
                    </div>
                    <div class="bg-blue-50 p-3 rounded-lg">
                        <div class="text-xs font-medium text-blue-600 uppercase tracking-wider">MAPE</div>
                        <div class="text-lg font-semibold text-blue-900" id="lstmMape">-</div>
                    </div>
                    <div class="bg-blue-50 p-3 rounded-lg">
                        <div class="text-xs font-medium text-blue-600 uppercase tracking-wider">Direction Accuracy</div>
                        <div class="text-lg font-semibold text-blue-900" id="lstmDirection">-</div>
                    </div>
                </div>
                
                <div class="mt-4 text-xs text-gray-500">
                    <p>Training Date: <span id="lstmTrainingDate">-</span></p>
                    <p>Test Samples: <span id="lstmTestSamples">-</span></p>
                </div>
            </div>

            <!-- SVR Metrics -->
            <div class="bg-white shadow rounded-lg p-6">
                <div class="flex items-center mb-4">
                    <div class="flex-shrink-0">
                        <i class="fas fa-cogs text-red-600 text-2xl"></i>
                    </div>
                    <div class="ml-3">
                        <h3 class="text-lg font-medium text-gray-900">Support Vector Regression</h3>
                        <p class="text-sm text-gray-500">Traditional ML Model</p>
                    </div>
                </div>
                
                <div class="grid grid-cols-2 gap-4">
                    <div class="bg-red-50 p-3 rounded-lg">
                        <div class="text-xs font-medium text-red-600 uppercase tracking-wider">RMSE</div>
                        <div class="text-lg font-semibold text-red-900" id="svrRmse">-</div>
                    </div>
                    <div class="bg-red-50 p-3 rounded-lg">
                        <div class="text-xs font-medium text-red-600 uppercase tracking-wider">R² Score</div>
                        <div class="text-lg font-semibold text-red-900" id="svrR2">-</div>
                    </div>
                    <div class="bg-red-50 p-3 rounded-lg">
                        <div class="text-xs font-medium text-red-600 uppercase tracking-wider">MAPE</div>
                        <div class="text-lg font-semibold text-red-900" id="svrMape">-</div>
                    </div>
                    <div class="bg-red-50 p-3 rounded-lg">
                        <div class="text-xs font-medium text-red-600 uppercase tracking-wider">Direction Accuracy</div>
                        <div class="text-lg font-semibold text-red-900" id="svrDirection">-</div>
                    </div>
                </div>
                
                <div class="mt-4 text-xs text-gray-500">
                    <p>Training Date: <span id="svrTrainingDate">-</span></p>
                    <p>Test Samples: <span id="svrTestSamples">-</span></p>
                </div>
            </div>
        </div>

        <!-- Performance Comparison Chart -->
        <div class="bg-white shadow rounded-lg p-6 mb-8">
            <h3 class="text-lg font-medium text-gray-900 mb-4">Performance Metrics Comparison</h3>
            <div class="relative">
                <canvas id="comparisonChart" width="400" height="300"></canvas>
            </div>
        </div>

        <!-- Winner Analysis -->
        <div id="winnerAnalysis" class="bg-white shadow rounded-lg p-6 mb-8">
            <h3 class="text-lg font-medium text-gray-900 mb-4">Model Performance Analysis</h3>
            <div id="analysisContent" class="prose max-w-none">
                <!-- Will be populated by JavaScript -->
            </div>
        </div>

        <!-- Model Parameters -->
        <div class="grid grid-cols-1 lg:grid-cols-2 gap-6">
            <!-- LSTM Parameters -->
            <div class="bg-white shadow rounded-lg p-6">
                <h3 class="text-lg font-medium text-gray-900 mb-4">LSTM Hyperparameters</h3>
                <div id="lstmParameters" class="space-y-2">
                    <!-- Will be populated by JavaScript -->
                </div>
            </div>

            <!-- SVR Parameters -->
            <div class="bg-white shadow rounded-lg p-6">
                <h3 class="text-lg font-medium text-gray-900 mb-4">SVR Hyperparameters</h3>
                <div id="svrParameters" class="space-y-2">
                    <!-- Will be populated by JavaScript -->
                </div>
            </div>
        </div>
    </div>

    <!-- Overall Model Performance -->
    <div class="bg-white shadow rounded-lg p-6 mt-8">
        <h3 class="text-lg font-medium text-gray-900 mb-4">Overall Model Performance Summary</h3>
        <div class="overflow-x-auto">
            <table class="min-w-full divide-y divide-gray-200">
                <thead class="bg-gray-50">
                    <tr>
                        <th class="px-6 py-3 text-left text-xs font-medium text-gray-500 uppercase tracking-wider">Stock</th>
                        <th class="px-6 py-3 text-left text-xs font-medium text-gray-500 uppercase tracking-wider">LSTM RMSE</th>
                        <th class="px-6 py-3 text-left text-xs font-medium text-gray-500 uppercase tracking-wider">SVR RMSE</th>
                        <th class="px-6 py-3 text-left text-xs font-medium text-gray-500 uppercase tracking-wider">LSTM R²</th>
                        <th class="px-6 py-3 text-left text-xs font-medium text-gray-500 uppercase tracking-wider">SVR R²</th>
                        <th class="px-6 py-3 text-left text-xs font-medium text-gray-500 uppercase tracking-wider">Better Model</th>
                    </tr>
                </thead>
                <tbody id="overallTable" class="bg-white divide-y divide-gray-200">
                    <!-- Will be populated by JavaScript -->
                </tbody>
            </table>
        </div>
    </div>
</div>
{% endblock %}

{% block extra_js %}
<script>
document.addEventListener('DOMContentLoaded', function() {
    loadOverallPerformance();
    
    // Event listeners
    document.getElementById('loadComparison').addEventListener('click', loadComparison);
});

let comparisonChart = null;

async function loadComparison() {
    const stock = document.getElementById('comparisonStock').value;
    
    if (!stock) {
        showNotification('Please select a stock', 'warning');
        return;
    }
    
    showLoading();
    
    try {
        const response = await fetch(`/api/comparison/${stock}/`);
        const data = await response.json();
        
        if (response.ok) {
            displayComparison(data);
            document.getElementById('comparisonResults').classList.remove('hidden');
        } else {
            showNotification(data.error || 'Failed to load comparison', 'error');
        }
        
    } catch (error) {
        console.error('Error loading comparison:', error);
        showNotification('Error loading comparison data', 'error');
    } finally {
        hideLoading();
    }
}

function displayComparison(data) {
    // Update LSTM metrics
    document.getElementById('lstmRmse').textContent = formatNumber(data.lstm_metrics.rmse, 4);
    document.getElementById('lstmR2').textContent = formatNumber(data.lstm_metrics.r2_score, 4);
    document.getElementById('lstmMape').textContent = formatPercent(data.lstm_metrics.mape);
    document.getElementById('lstmDirection').textContent = formatPercent(data.lstm_metrics.direction_accuracy);
    document.getElementById('lstmTrainingDate').textContent = new Date(data.lstm_metrics.training_date).toLocaleDateString();
    document.getElementById('lstmTestSamples').textContent = data.lstm_metrics.test_samples;
    
    // Update SVR metrics
    document.getElementById('svrRmse').textContent = formatNumber(data.svr_metrics.rmse, 4);
    document.getElementById('svrR2').textContent = formatNumber(data.svr_metrics.r2_score, 4);
    document.getElementById('svrMape').textContent = formatPercent(data.svr_metrics.mape);
    document.getElementById('svrDirection').textContent = formatPercent(data.svr_metrics.direction_accuracy);
    document.getElementById('svrTrainingDate').textContent = new Date(data.svr_metrics.training_date).toLocaleDateString();
    document.getElementById('svrTestSamples').textContent = data.svr_metrics.test_samples;
    
    // Create comparison chart
    createComparisonChart(data.comparison_chart_data);
    
    // Update parameters
    updateParameters(data.lstm_metrics.model_parameters, 'lstmParameters');
    updateParameters(data.svr_metrics.model_parameters, 'svrParameters');
    
    // Generate analysis
    generateAnalysis(data);
}

function createComparisonChart(chartData) {
    const ctx = document.getElementById('comparisonChart').getContext('2d');
    
    if (comparisonChart) {
        comparisonChart.destroy();
    }
    
    comparisonChart = new Chart(ctx, {
        type: 'radar',
        data: {
            labels: chartData.metrics,
            datasets: [
                {
                    label: 'LSTM',
                    data: chartData.lstm_values,
                    borderColor: 'rgb(59, 130, 246)',
                    backgroundColor: 'rgba(59, 130, 246, 0.2)',
                    pointBackgroundColor: 'rgb(59, 130, 246)',
                    pointBorderColor: '#fff',
                    pointHoverBackgroundColor: '#fff',
                    pointHoverBorderColor: 'rgb(59, 130, 246)'
                },
                {
                    label: 'SVR',
                    data: chartData.svr_values,
                    borderColor: 'rgb(239, 68, 68)',
                    backgroundColor: 'rgba(239, 68, 68, 0.2)',
                    pointBackgroundColor: 'rgb(239, 68, 68)',
                    pointBorderColor: '#fff',
                    pointHoverBackgroundColor: '#fff',
                    pointHoverBorderColor: 'rgb(239, 68, 68)'
                }
            ]
        },
        options: {
            responsive: true,
            maintainAspectRatio: false,
            scales: {
                r: {
                    beginAtZero: true,
                    angleLines: {
                        display: true
                    },
                    grid: {
                        circular: true
                    }
                }
            },
            plugins: {
                title: {
                    display: true,
                    text: 'Model Performance Radar Chart'
                },
                legend: {
                    position: 'top'
                }
            }
        }
    });
}

function updateParameters(parameters, containerId) {
    const container = document.getElementById(containerId);
    container.innerHTML = '';
    
    if (parameters && typeof parameters === 'object') {
        for (const [key, value] of Object.entries(parameters)) {
            const div = document.createElement('div');
            div.className = 'flex justify-between py-1 border-b border-gray-200';
            div.innerHTML = `
                <span class="text-sm font-medium text-gray-600">${key.replace(/_/g, ' ').toUpperCase()}</span>
                <span class="text-sm text-gray-900">${value}</span>
            `;
            container.appendChild(div);
        }
    } else {
        container.innerHTML = '<p class="text-sm text-gray-500">No parameters available</p>';
    }
}

function generateAnalysis(data) {
    const lstmMetrics = data.lstm_metrics;
    const svrMetrics = data.svr_metrics;
    
    let analysis = '<div class="space-y-4">';
    
    // Compare RMSE (lower is better)
    if (lstmMetrics.rmse < svrMetrics.rmse) {
        const improvement = (((svrMetrics.rmse - lstmMetrics.rmse) / svrMetrics.rmse) * 100).toFixed(2);
        analysis += `<p class="text-green-600"><strong>✓ LSTM shows better accuracy</strong> with ${improvement}% lower RMSE than SVR.</p>`;
    } else {
        const improvement = (((lstmMetrics.rmse - svrMetrics.rmse) / lstmMetrics.rmse) * 100).toFixed(2);
        analysis += `<p class="text-red-600"><strong>✓ SVR shows better accuracy</strong> with ${improvement}% lower RMSE than LSTM.</p>`;
    }
    
    // Compare R² (higher is better)
    if (lstmMetrics.r2_score > svrMetrics.r2_score) {
        const improvement = ((lstmMetrics.r2_score - svrMetrics.r2_score) * 100).toFixed(2);
        analysis += `<p class="text-green-600"><strong>✓ LSTM explains more variance</strong> with ${improvement} points higher R² score.</p>`;
    } else {
        const improvement = ((svrMetrics.r2_score - lstmMetrics.r2_score) * 100).toFixed(2);
        analysis += `<p class="text-red-600"><strong>✓ SVR explains more variance</strong> with ${improvement} points higher R² score.</p>`;
    }
    
    // Compare direction accuracy
    if (lstmMetrics.direction_accuracy > svrMetrics.direction_accuracy) {
        const improvement = (lstmMetrics.direction_accuracy - svrMetrics.direction_accuracy).toFixed(2);
        analysis += `<p class="text-green-600"><strong>✓ LSTM better predicts price direction</strong> with ${improvement}% higher accuracy.</p>`;
    } else {
        const improvement = (svrMetrics.direction_accuracy - lstmMetrics.direction_accuracy).toFixed(2);
        analysis += `<p class="text-red-600"><strong>✓ SVR better predicts price direction</strong> with ${improvement}% higher accuracy.</p>`;
    }
    
    // Overall recommendation
    const lstmScore = (lstmMetrics.r2_score * 2) + (100 - lstmMetrics.mape) + lstmMetrics.direction_accuracy - (lstmMetrics.rmse * 10);
    const svrScore = (svrMetrics.r2_score * 2) + (100 - svrMetrics.mape) + svrMetrics.direction_accuracy - (svrMetrics.rmse * 10);
    
    if (lstmScore > svrScore) {
        analysis += `<div class="bg-green-50 border border-green-200 rounded-lg p-4">
            <p class="text-green-800"><strong>🏆 Recommendation: LSTM Neural Network</strong></p>
            <p class="text-green-700 text-sm mt-1">LSTM shows superior overall performance for this stock based on combined metrics.</p>
        </div>`;
    } else {
        analysis += `<div class="bg-red-50 border border-red-200 rounded-lg p-4">
            <p class="text-red-800"><strong>🏆 Recommendation: Support Vector Regression</strong></p>
            <p class="text-red-700 text-sm mt-1">SVR shows superior overall performance for this stock based on combined metrics.</p>
        </div>`;
    }
    
    analysis += '</div>';
    
    document.getElementById('analysisContent').innerHTML = analysis;
}

async function loadOverallPerformance() {
    try {
        const response = await fetch('/api/metrics/');
        const data = await response.json();
        
        if (data.results && data.results.length > 0) {
            updateOverallTable(data.results);
        }
        
    } catch (error) {
        console.error('Error loading overall performance:', error);
    }
}

function updateOverallTable(metrics) {
    const tableBody = document.getElementById('overallTable');
    tableBody.innerHTML = '';
    
    // Group metrics by stock
    const stockMetrics = {};
    
    metrics.forEach(metric => {
        if (!stockMetrics[metric.stock_symbol]) {
            stockMetrics[metric.stock_symbol] = {};
        }
        stockMetrics[metric.stock_symbol][metric.model_type] = metric;
    });
    
    // Create table rows
    for (const [stock, models] of Object.entries(stockMetrics)) {
        if (models.LSTM && models.SVR) {
            const betterModel = models.LSTM.rmse < models.SVR.rmse ? 'LSTM' : 'SVR';
            
            const row = tableBody.insertRow();
            row.innerHTML = `
                <td class="px-6 py-4 whitespace-nowrap text-sm font-medium text-gray-900">${stock}</td>
                <td class="px-6 py-4 whitespace-nowrap text-sm text-gray-500">${formatNumber(models.LSTM.rmse, 4)}</td>
                <td class="px-6 py-4 whitespace-nowrap text-sm text-gray-500">${formatNumber(models.SVR.rmse, 4)}</td>
                <td class="px-6 py-4 whitespace-nowrap text-sm text-gray-500">${formatNumber(models.LSTM.r2_score, 4)}</td>
                <td class="px-6 py-4 whitespace-nowrap text-sm text-gray-500">${formatNumber(models.SVR.r2_score, 4)}</td>
                <td class="px-6 py-4 whitespace-nowrap text-sm font-medium ${betterModel === 'LSTM' ? 'text-blue-600' : 'text-red-600'}">${betterModel}</td>
            `;
        }
    }
    
    if (tableBody.children.length === 0) {
        const row = tableBody.insertRow();
        row.innerHTML = '<td colspan="6" class="px-6 py-4 text-center text-gray-500">No model comparisons available</td>';
    }
}
</script>
{% endblock %}